import argparse
from ipex_models.qwen2_chat_xpu import Qwen2ChatModel


def main(prompt):
    model = Qwen2ChatModel(model_name="Qwen/Qwen2.5-7B-Instruct",local_dir="/models/Qwen/Qwen2.5-7B-Instruct")
    response = model.chat(prompt)
    print(response)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Qwen2.5-Chat")
    parser.add_argument("--prompt", required=True, help="Prompt for the model")
    args = parser.parse_args()
    main(args.prompt)
